package 链表;

import java.util.ArrayList;
import java.util.List;

/**
 * 输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。
 * <p>
 * 例如，一个链表有 6 个节点，从头节点开始，它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。
 * <p>
 * 示例：
 * <p>
 * 给定一个链表: 1->2->3->4->5, 和 k = 2.
 * <p>
 * 返回链表 4->5.
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/lian-biao-zhong-dao-shu-di-kge-jie-dian-lcof
 */
public class 剑指_Offer_22_链表中倒数第k个节点 {

    public static void main(String[] args) {
        ListNode listNode1 = new ListNode(1, new ListNode(2, new ListNode(4, null)));
        System.out.println(getKthFromEnd(listNode1, 2));
        System.out.println(getKthFromEnd1(listNode1, 2));
        System.out.println(getKthFromEndByDoublePoint(listNode1, 2));
    }

    public static ListNode getKthFromEnd(ListNode head, int k) {
        if (head == null || k < 0) return null;

        ArrayList<ListNode> listNodes = new ArrayList<>();
        ListNode temp = head;
        while (temp != null) {
            listNodes.add(temp);
            temp = temp.next;
        }

        return listNodes.get(listNodes.size() - k);

    }

    public static ListNode getKthFromEnd1(ListNode head, int k) {
        if (head == null || k < 0) return null;

        ListNode temp = head;
        int len = 0;
        while (temp != null) {
            len++;
            temp = temp.next;
        }
        temp = head;
        for (int i = 0; i < len - k; i++) {
            temp = temp.next;
        }

        return temp;

    }

    //双指针
    public static ListNode getKthFromEndByDoublePoint(ListNode head, int k) {
        if (head == null || k < 0) return null;

        ListNode slow = head, fast = head;

//        for (int i = 0; i < k; i++) {
//            fast = fast.next;
//        }

        while(k-- > 0){
            fast = fast.next;
        }

        while (fast != null) {
            fast = fast.next;
            slow = slow.next;
        }
        return slow;
    }

}
